How to make a Java TreeMap be sorted by some comparator